import sys

from downloader import Downloader


def GetCMDArgs(key, default):
  if key not in sys.argv:
    return default
  return sys.argv[sys.argv.index(key) + 1]


def GetBool(s):
  if s == '0':
    return False
  return True


if __name__ == '__main__':
  if '-h' in sys.argv:
    print('''
    python3 main.py -n 小说名字 -u 目录URL -c 抓取使用的正则表达式配置文件路径 [-s 1或0，是否额外分章保存]  
    根据指定书籍目录页面爬取整本小说  
    -n 默认值：明克街13号  
    -u 默认值：https://www.mingkejie13hao.com/book/31532/  
    -c 默认值：mingkejie13hao.json  
      正则表达式配置文件特别说明：列表存放的正则表达式除最后一项外，之前的表达式只会使用匹配到的第0个结果，最后一项会使用所有结果。  
    -s 默认值：0（不额外分章保存）
    
    python3 main.py -h  
    获取帮助信息  
    ''')
    exit(0)
  downloader = Downloader(
    GetCMDArgs('-n', '明克街13号'),
    GetCMDArgs('-u', 'https://www.mingkejie13hao.com/book/31532/'),
    GetCMDArgs('-c', 'mingkejie13hao.json'),
    GetBool(GetCMDArgs('-s', '0')),
  )
  if downloader.configLoaded:
    downloader.Run()
